Small fixes to test_dropout.py