Adds back missing test