fix forgotten tests