fixing tests i broke in previous commit