Fixing dropout so tests pass again