Restore dev dependency used for unit tests

Signed-off-by: Aran Wilkinson <aran3001@gmail.com>