Fixed jest integration