Added missing deps
Updated to end test at right time