minor fix
bugfix + added test