Changed style of assertion in test