Cleanup

Signed-off-by: Jeremy Massel <jeremy@habithq.ca>