Clean up && fix build
Signed-off-by: PipelinerArd <kristoferard@gmail.com>