Fixed bug in test