refactored test
added assertion + refactored test