How to setup Behave/Python Step Definitions in Intellij 2017.1

Something changed when I upgraded to 2017.1 from 2016.  Before, IntelliJ understood how to link my step definitions to my .feature files for Python/Behave.

Now, I can't track down how to get IntelliJ to recognize my step definitions.  I've tried adding the steps directory to the project definitions as a module, Sources directory, Annotations, etc.  I've googled ad nauseum to try and find a solution.

Directory Setup:

--tests/
----features/
-------feature1-feature-files/
-------feature2-feature-files/
----steps/

Anybody know how to do this?

IntelliJ IDEA 2017.1
Build #IU-171.3780.107, built on March 22, 2017
Licensed to Zipit Wireless, Inc. / Mike Crowe
Subscription is active until January 6, 2018
JRE: 1.8.0_112-release-736-b13 amd64
JVM: OpenJDK 64-Bit Server VM by JetBrains s.r.o
Linux 4.4.0-72-generic