Forgot some testing code