#281 fixing tests