Fixed so build step works. Probably need more work here.