PR Tests: fixed missing import