Added missing test from last import