fix import error in tests
Merge pull request #4 from RaphaelKimmig/design_v4

fix import error in tests