bugfix to logic
fixing imports