minor changes
play with types in test