Fixed missing test from repo