Forgot some tests