Fix the flash

@square/railsfork-contributors apparently the flash is totally broken on 4.0 =/